package dao

import (
	"shorten/internal/model"
	"shorten/pkg/util"
	"strings"
)

func GetOneUrl(sign string) string {
	var url string
	err := DB().Model(&model.Shorten{}).Select("base_url").Where("short_sign = ?", sign).Limit(1).Find(&url).Error
	util.CheckErr(err)
	return url
}

func GetOneShorten(where map[string]any, fields []string) *model.Shorten {
	var data model.Shorten
	err := DB().Model(&model.Shorten{}).Select(strings.Join(fields, ",")).Where(where).Limit(1).Find(&data).Error
	util.CheckErr(err)
	return &data
}
